from argparse import Namespace
import torch
import time

from pytorch_lightning.loggers import NeptuneLogger



config = Namespace(

    AVAIL_GPUS = min(1, torch.cuda.device_count()),
    BATCH_SIZE = 256 if min(1, torch.cuda.device_count()) else 64,
    lr = 0.02,
    optim_type = "Adam",
    max_epochs = 3,
    ckpt_path ="best.ckpt",
    data_dir ="../../../../data/",
    dropout_p = 0.1
)

logger = NeptuneLogger(

    api_key = "eyJhcGlfYWRkcmVzcyI6Imh0dHBzOi8vYXBwLm5lcHR1bmUuYWkiLCJhcGlfdXJsIjoiaHR0cHM6Ly9hcHAubmVwdHVuZS5haSIsImFwaV9rZXkiOiIxNGEwOWZhYS1iM2IwLTQ3ZjItYmE2MS02ZjVjOTNjMTcyZDcifQ==",
# api_key="ANONYMOUS",
    project = 'shao1chuan/test1',
    name = time.strftime("%Y_%m_%d_%H_%M_%S", time.localtime(time.time())),
tags=["simple", "showcase"],
    log_model_checkpoints = False,
    prefix= "training")


